package cn.myworld.algorithm.linear;

import cn.myworld.algorithm.common.BaseList;

import java.util.Iterator;

/**
 * 双向链表
 * @param <T>
 */
public class TowWayLinkList<T> implements BaseList<T> {

    private Node head;
    private Node last;
    private int N;

    private class Node{
        public Node(T data, Node pre, Node next){
            this.data = data;
            this.pre = pre;
            this.next = next;
        }

        public T data;
        // 指向上一个节点
        public Node pre;
        // 指向下一个节点
        public Node next;
    }

    public TowWayLinkList() {
        // 初始化头结点和尾节点
        this.head = new Node(null,null,null);
        this.last = null;
        // 初始化元素个数
        this.N = 0;
    }

    @Override
    public void clear() {
        this.head.next = null;
        this.head.pre = null;
        this.head.data = null;
        this.last = null;
        this.N = 0;
    }

    @Override
    public boolean isEmpty() {
        return N == 0;
    }

    @Override
    public int length() {
        return N;
    }

    public T getFirst() {
        if (isEmpty()){
            return null;
        }
        return head.next.data;
    }

    public T getLast() {
        if (isEmpty()){
            return null;
        }
        return last.data;
    }

    @Override
    public T get(int i) {
       Node n =  head.next;
       for (int index = 0; index < i; index++){
           n = n.next;
       }
       return n.data;
    }

    @Override
    public void insert(T e) {
        if (isEmpty()){
            // 如果链表为空
            // 创建新的节点
            Node newNode = new Node(e, head, null);
            // 让新节点成为尾节点
            last = newNode;
            // 让头结点指向尾节点
            head.next = last;
        } else {
            // 如果链表不为空
            Node oldLast = last;
            // 创建新节点
            Node newNode = new Node(e, oldLast, null);
            // 让当前尾节点指向新节点
            oldLast.next = newNode;
            // 让新节点成为尾节点
            last = newNode;
        }
        // 元素个数+1
        N++;
    }

    @Override
    public void insert(int i, T e) {
        // 找到i位置的前节点
        Node pre = head;
        for (int index = 0; index < i; index++) {
            pre = pre.next;
        }
        // 找到i位置的节点
        Node cur = pre.next;
        // 创建新节点
        Node newNode = new Node(e, pre, cur);
        // 让i位置的前一个节点的下一个节点变为新节点
        pre.next = newNode;
        // 让i位置的前节点变为新节点
        cur.pre = newNode;
        // 元素个数+1
        N++;
    }

    @Override
    public T remove(int i) {
        // 找到i位置的前一个节点
        Node pre = head;
        for (int index = 0; index < i; index++){
            pre = pre.next;
        }
        // 找到i位置的节点
        Node cur = pre.next;
        // 找到i位置的下一个节点
        Node next = cur.next;
        // 让i位置的前一个节点的下一个节点指向i位置的下一个节点
        pre.next = next;
        // 让i位置的下一个节点的前节点指向i位置的前一个节点
        next.pre = pre;
        // 元素个数-1
        N--;
        return cur.data;
    }

    @Override
    public int indexOf(T e) {
        Node n = head;
        for (int i = 0; n.next != null; i++){
            n = n.next;
            if (n.data == e){
                return i;
            }
        }
        return -1;
    }

    @Override
    public Iterator<T> iterator() {
        return new TIterator();
    }

    private class TIterator implements Iterator<T> {

        private Node n;
        public TIterator() {
            n = head;
        }

        @Override
        public boolean hasNext() {
            return n.next != null;
        }

        @Override
        public T next() {
            n = n.next;
            return n.data;
        }
    }
}
